Skip to content

[WIP] Add YuE model - #41041

Closed
Manalelaidouni wants to merge 6 commits into
huggingface:mainfrom
Manalelaidouni:add-yue-model
Closed

Manalelaidouni wants to merge 6 commits into
huggingface:mainfrom
Manalelaidouni:add-yue-model

Conversation

@Manalelaidouni

Copy link
Copy Markdown
Contributor

What does this PR do?

This draft PR aims at integrating the YuE model to transformers, it's a lyrics to song generation model that takes a lyrics prompt and a reference audio tokenized using a finetuned Xcodec model, YuE itself uses a two stage generation based on 2 Llama2 models, toped with the Vocos vocoder to enhance the produced song.

Progress : the YuEProcessor and YuETokenizer produce identical results as the original YuE implementation and I'm currently working on the replicating modeling part.

@Rocketknight1

Copy link
Copy Markdown
Member

cc @eustlb @ebezzam

@ArthurZucker

Copy link
Copy Markdown
Collaborator

Thanks for the work on YuE @Manalelaidouni! This draft hasn't moved in a long time and the modeling part never landed, so I'm closing it for now. Please feel free to reopen or start a fresh modular PR if you get back to it.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants